import tensorflow as tf
import matplotlib as mat


def psnr(tf_img1, tf_img2):
    return tf.image.psnr(tf_img1, tf_img2, max_val=255)


def ssim(tf_img1, tf_img2):
    tf.image.ssim(tf_img1, tf_img2, max_val=255)

